from flask import g

"""
自定义一个装饰器，验证是否登录，没有登录就拦截
"""


def login_required(func):
    def wrapper(*args, **kwargs):
        if g.user_id is not None:  # 已经登录
            return func(*args, **kwargs)
        else:
            return {'msg': '没有登录'}

    return wrapper
